package com.example.springboottestgit.controller.main;

import cn.hutool.http.HttpUtil;
import com.alibaba.fastjson.JSONObject;
import com.example.springboottestgit.bean.vo.InputVO;
import com.example.springboottestgit.bean.vo.OutputVO;

import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * @description:
 * @author: ZZZ
 */
public class Test {

    static final int a = 5;
    static final List<String> list = new ArrayList<>();

    public static void main(String[] args) throws IOException {



        List<String> list = new ArrayList<>();

        //list.add("2024年3月中国海油的A股股价最高是多少");

//        list.add("2024年中国海油（600938）最高股价是多少");
//        list.add("2024年中国海油（600938）最低股价是多少");
//        list.add("2024年中国海油（600938）最高股价是多少");
//        list.add("2024年中国海油（600938）最低股价是多少");
//        list.add("2024年中国海油（600938）最高股价是多少");
//        list.add("2024年中国海油（600938）最低股价是多少");


        list.add("2024年中国石油（601857）最高股价是多少");
        list.add("2024年中国石油（601857）最低股价是多少");



 /*       String path = "D:/file/qwen2-test.txt";
        FileReader reader = new FileReader(path);
        BufferedReader bufferedReader = new BufferedReader(reader);
        StringBuilder stringBuilder = new StringBuilder();
        String line;

        while ((line = bufferedReader.readLine()) != null) {
            //stringBuilder.append(line);

            list.add(line);
        }
        bufferedReader.close();*/


        for (String question : list) {
            System.out.println("====================================================");
            System.out.println(question);
            String answer = askLawQuestion(question);
            System.out.println(answer);

        }

    }

    public static String askLawQuestion(String question) {
        //String url = "http://192.168.80.35:19327/v1/completions";
        String url = "http://192.168.80.35:8091/qwen2/api";
        InputVO inputVO = new InputVO();
        inputVO.setQuestion(question);
        inputVO.setFilePath("");
        String param = JSONObject.toJSONString(inputVO);
        String result = HttpUtil.createPost(url)
                //这个请求头.header是自己项目需要加的，可以省略
                .header("Content-Type", "application/json")
                //这两个请求头是项目需要加的，可以省略
                //传输参数
                .body(param).execute().body();
        OutputVO out = JSONObject.parseObject(result, OutputVO.class);
        String text = out.getAnswer();
        try {
            Thread.sleep(1000);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
        return text;
    }
}
